#pragma once
#include <iostream>
#include <string>

#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>

using namespace std;

class InetAddr
{
private:
    void GetAddress(string* ip,uint16_t* port)
    {
        *port = ntohs(_addr.sin_port);
        *ip = inet_ntoa(_addr.sin_addr);

    }
public:
    InetAddr(const struct sockaddr_in addr)
    :_addr(addr)
    {
        GetAddress(&_ip,&_port);
    }

    //_localaddr("0",8888);
    InetAddr(const string ip,uint16_t port)
    :_ip(ip)
    ,_port(port)
    {
       _addr.sin_family=AF_INET;
       _addr.sin_port=htons(_port);
       _addr.sin_addr.s_addr=inet_addr(_ip.c_str());
    }

    InetAddr()
    {}

    bool operator==(const InetAddr addr)
    {
        if(_ip==addr._ip&&_port==addr._port)
        {
            return true;
        }
        else 
        {
            return false;
        }
    }

    void operator=(struct sockaddr_in addr)
    {
        _addr=addr;
        _ip=inet_ntoa(addr.sin_addr);
        _port=addr.sin_port;
    }

    string IP()
    {
        return _ip;
    }

    uint16_t Port()
    {
        return _port;
    }
    struct sockaddr_in Addr()
    {
        return _addr;
    }

private:
    struct sockaddr_in _addr;
    string _ip;
    uint16_t _port;
};